package risk.game.entity.unit;

import java.util.ArrayList;
import java.util.List;

import risk.game.entity.player.Player;

public class Army {

	private List<Cannon>  cannonUnits;
	private List<Soldier> soldierUnits;
	private List<Trooper> trooperUnits;
	private Player ownerPlayer;
	
	public Army(Player ownerPlayer){
		this.ownerPlayer = ownerPlayer;
	}
	
	public List<Cannon> getCannonUnits() {
		return cannonUnits;
	}
	public void setCannonUnits(List<Cannon> cannonUnits) {
		this.cannonUnits = cannonUnits;
	}
	public List<Soldier> getSoldierUnits() {
		return soldierUnits;
	}
	public void setSoldierUnits(List<Soldier> soldierUnits) {
		this.soldierUnits = soldierUnits;
	}
	public List<Trooper> getTrooperUnits() {
		return trooperUnits;
	}
	public void setTrooperUnits(List<Trooper> trooperUnits) {
		this.trooperUnits = trooperUnits;
	}
	public Player getOwnerPlayer() {
		return ownerPlayer;
	}
	public void setOwnerPlayer(Player ownerPlayer) {
		this.ownerPlayer = ownerPlayer;
	}
	
	public void addCannon(Cannon cannon){
		if(cannonUnits == null)
			cannonUnits = new ArrayList<Cannon>();
		cannonUnits.add(cannon);
	}
	public void addSoldier(Soldier soldier){
		if(soldierUnits == null)
			soldierUnits = new ArrayList<Soldier>();
		soldierUnits.add(soldier);
	}
	public void addTrooper(Trooper trooper){
		if(trooperUnits == null)
			trooperUnits = new ArrayList<Trooper>();
		trooperUnits.add(trooper);
	}
	
	
}
